import unittest

from count import count


class TestAdd(unittest.TestCase):
    def test_add1(self):
        c=count(23,12)
        r=c.add()
        self.assertEqual(r, 35)

    def test_add2(self):
        c=count(2,12)
        r=c.add()
        self.assertEqual(r, 14)

    def test_add3(self):
        c=count(2,3)
        r=c.add()
        self.assertEqual(r, 5)


if __name__ == '__main__':
    unittest.main()
